import torch
from torch import nn

# e.g. sequential
model = nn.Sequential(
    nn.Conv2d(1, 1, 3, padding=1),
    nn.Linear(784, 256),
    nn.ReLU(),
    nn.Linear(256, 10)
)